

import java.io.File;
import java.io.FileOutputStream;
import java.io.FileWriter;
import java.io.OutputStreamWriter;
import java.nio.charset.Charset;
import java.nio.charset.CharsetEncoder;
import java.util.Collection;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.apache.commons.io.FileUtils;
import org.dom4j.Attribute;
import org.dom4j.Document;
import org.dom4j.Element;
import org.dom4j.io.OutputFormat;
import org.dom4j.io.SAXReader;
import org.dom4j.io.XMLWriter;
import org.junit.Test;

public class ModifyTxnXML {
	
	public static void main(String[] args) {
  	  File files1=new File("E:/workspace_dev/ngbc");
			Collection<File> filesCollection1 = FileUtils.listFiles(files1, new String[] { "txn.xml.bak.xml"}, true);
			if (filesCollection1.size() >0 && filesCollection1 !=null) {
				for (File file : filesCollection1) {
					System.out.println(file.getAbsolutePath());
					file.delete();  
				}
			}
	}
	
	
     Map dict=new HashMap<>();
	@Test
	public void test() throws Exception {
		// 创建saxReader对象
		SAXReader reader = new SAXReader();
		Map dicthz=new HashMap<>();
		
  	    File files1=new File("E:/type");
		Collection<File> filesCollection1 = FileUtils.listFiles(files1, new String[] { "dataDictionary.xml"}, true);
		if (filesCollection1.size() >0 && filesCollection1 !=null) {
			for (File file : filesCollection1) {
				System.out.println(file.getAbsolutePath());
				String na=file.getName().replaceAll(".dataDictionary.xml", "");
				System.out.println(na);
				// 通过read方法读取一个文件 转换成Document对象
				Document document = reader.read(file);
				//获取根节点元素对象
				Element node = document.getRootElement();
				dict=new HashMap<>();
				//遍历数据字典
				listNodesDict(node);
				dicthz.put(na, dict);
			}
		}
			//System.out.println(dicthz);		
//		Set<String> keys = dict.keySet();
//		for(String key :keys){
//		     System.out.println(key+" "+dict.get(key));
//		}
		//dicthz
	  	    File files=new File("E:/workspace_dev/ngbc");
			Collection<File> filesCollection = FileUtils.listFiles(files, new String[] { "txn.xml"}, true);
			if (filesCollection.size() >0 && filesCollection1 !=null) {
				for (File file : filesCollection) {
					System.out.println(file.getAbsolutePath());
				 String type=file.getAbsolutePath().split("\\\\")[3];
				 System.out.println(type);
					//判断文件位置
				    dict=new HashMap<>();
					dict=(Map) dicthz.get("gl");
					if(dict!=null){
						Document document1 = reader.read(file);
						//获取根节点元素对象
						Element node1 = document1.getRootElement();
						//遍历所有的元素节点
						listNodesTXN(node1);
						writer(document1,file.getAbsolutePath());
					}

				
				}
			}		
		
		

		
		
		//遍历所有的元素节点
		//listNodes(node);

//		// 获取四大名著元素节点中，子节点名称为红楼梦元素节点。
//		Element element = node.element("红楼梦");
//		//获取element的id属性节点对象
//		Attribute attr = element.attribute("id");
//		//删除属性
//		element.remove(attr);
//		//添加新的属性
//		element.addAttribute("name", "作者");
//		// 在红楼梦元素节点中添加朝代元素的节点
//		Element newElement = element.addElement("朝代");
//		newElement.setText("清朝");
//		//获取element中的作者元素节点对象
//		Element author = element.element("作者");
//		//删除元素节点
//		boolean flag = element.remove(author);
//		//返回true代码删除成功，否则失败
//		System.out.println(flag);
//		//添加CDATA区域
//		element.addCDATA("红楼梦，是一部爱情小说.");
//		// 写入到一个新的文件中
		

	}

	/**
	 * 把document对象写入新的文件
	 * 
	 * @param document
	 * @param filename 
	 * @throws Exception
	 */
	public void writer(Document document, String filename) throws Exception {
		// 紧凑的格式
		// OutputFormat format = OutputFormat.createCompactFormat();
		// 排版缩进的格式
		OutputFormat format = OutputFormat.createPrettyPrint();
		// 设置编码
		format.setEncoding("UTF-8");
		// 创建XMLWriter对象,指定了写出文件及编码格式
		// XMLWriter writer = new XMLWriter(new FileWriter(new
		// File("src//a.xml")),format);
		XMLWriter writer = new XMLWriter(new OutputStreamWriter(
				new FileOutputStream(new File(filename)), "UTF-8"), format);
		// 写入
		writer.write(document);
		// 立即写入
		writer.flush();
		// 关闭操作
		writer.close();
	}

	/**
	 * 遍历当前节点元素下面的所有(元素的)子节点
	 * 
	 * @param node
	 */
	public void listNodes(Element node) {
		System.out.println("当前节点的名称：：" + node.getName());
		// 获取当前节点的所有属性节点
		List<Attribute> list = node.attributes();
		// 遍历属性节点
		
		
		for (Attribute attr : list) {
			System.out.println(attr.getText() + "-----" + attr.getName()
					+ "---" + attr.getValue());
		}

		if (!(node.getTextTrim().equals(""))) {
			System.out.println("文本内容：：：：" + node.getText());
		}

		// 当前节点下面子节点迭代器
		Iterator<Element> it = node.elementIterator();
		// 遍历
		while (it.hasNext()) {
			// 获取某个子节点对象
			Element e = it.next();
			// 对子节点进行遍历
			listNodes(e);
		}
	}

	/**
	 * 介绍Element中的element方法和elements方法的使用
	 * 
	 * @param node
	 */
	public void elementMethod(Element node) {
		// 获取node节点中，子节点的元素名称为西游记的元素节点。
		Element e = node.element("西游记");
		// 获取西游记元素节点中，子节点为作者的元素节点(可以看到只能获取第一个作者元素节点)
		Element author = e.element("作者");

		System.out.println(e.getName() + "----" + author.getText());

		// 获取西游记这个元素节点 中，所有子节点名称为作者元素的节点 。

		List<Element> authors = e.elements("作者");
		for (Element aut : authors) {
			System.out.println(aut.getText());
		}

		// 获取西游记这个元素节点 所有元素的子节点。
		List<Element> elements = e.elements();

		for (Element el : elements) {
			System.out.println(el.getText());
		}

	}
	
	
	/**
	 * 读取数据字典
	 * 
	 * @param node
	 */
	public void listNodesDict(Element node) {
		System.out.println("当前节点的名称：：" + node.getName());
		// 获取当前节点的所有属性节点
		List<Attribute> list = node.attributes();
		// 遍历属性节点
		
		if(node.getName().equals("element")){
			
			Attribute attr1 = node.attribute("id");
			System.out.println("测试一下是否是id"+attr1.getValue());
//			dict
			Map dictnode =new HashMap<String,String>();
			for (Attribute attr : list) {
				System.out.println(attr.getText() + "-----" + attr.getName()
						+ "---" + attr.getValue());
				dictnode.put(attr.getName(), attr.getValue());
				
			}
			dict.put(attr1.getValue(), dictnode);
		}
		

		if (!(node.getTextTrim().equals(""))) {
			System.out.println("文本内容：：：：" + node.getText());
		}

		// 当前节点下面子节点迭代器
		Iterator<Element> it = node.elementIterator();
		// 遍历
		while (it.hasNext()) {
			// 获取某个子节点对象
			Element e = it.next();
			// 对子节点进行遍历
			listNodesDict(e);
		}
	}
	
	
	
	
	/**
	 * 遍历当前节点元素下面的所有(元素的)子节点
	 * 
	 * @param node
	 */
	public void listNodesTXN(Element node) {
		System.out.println("当前节点的名称：：" + node.getName());
		// 获取当前节点的所有属性节点
		List<Attribute> list = node.attributes();
		// 遍历属性节点
		
		if(node.getName().equals("field")){
			Attribute attr1 = node.attribute("id");
			System.out.println("测试一下是否是id："+attr1.getValue());
		    String fieldname=attr1.getValue();
			for (Attribute attr : list) {
				System.out.println(attr.getText() + "-----" + attr.getName()
						+ "---" + attr.getValue());
				
				System.out.println("输出什么？："+attr.getName() );
				try {
					String value=(String) ((Map<?, ?>)dict.get(fieldname)).get(attr.getName());
					   System.out.println(attr.getName()+"值："+value);
					   if(value!=null)
						   attr.setValue(value);
					   else {
						   if(attr.getName().equals("enumTypeId"))
							   attr.setValue("");
					   }

				} catch (Exception e) {
					// TODO: handle exception
					 System.out.println(attr.getName()+"值异常："+fieldname);
				}
			   
//				if(attr.getValue().equals("corp_cd")){
//					attr.setValue("sdfsdfsdf");
//				}
				
			}
			
			
		
		}
		
		

		if (!(node.getTextTrim().equals(""))) {
			System.out.println("文本内容：：：：" + node.getText());
		}

		// 当前节点下面子节点迭代器
		Iterator<Element> it = node.elementIterator();
		// 遍历
		while (it.hasNext()) {
			// 获取某个子节点对象
			Element e = it.next();
			// 对子节点进行遍历
			listNodesTXN(e);
		}
	}

}